from Page.basepage import BasePage
from Page.login import Login
from Page.homepage import HomePage


# 该类为进入不同页面（对<我的，首页，课程，直播，联系>五个页面建模page的调配page
class Main(BasePage):

    def quit(self):
        self._driver.quit()

    def go_my_login(self):  # 进入我的
        self.steps('/TestData/main.yaml', 'go_my_login')
        return Login(self._driver)

    def go_my_homepage(self):  # 进入主页
        self.steps('/TestData/main.yaml', 'go_homepage')
        return HomePage(self._driver)

    # def go_homepage(self):  # 进入首页
    #     if self.isElementPresent("id", "com.zikao.eduol:id/et_search") is True:
    #         print('----当前已在书城页面-----')
    #         return BooksMall(self._driver)
    #     else:
    #         self.steps('/TestData/main.yaml', 'go_homepage')
    #         print('----点击打开-首页-页面----')
    #         return BooksMall(self._driver)
    #
    # def go_Elective(self):  # 进入选课
    #     self.steps('/TestData/main.yaml', 'go_Elective')
    #     print('----点击打开-选课-页面----')
    #     return Login(self._driver)



